package com.movie.repository;

import com.movie.entity.UserRating;
import com.movie.entity.User;
import com.movie.entity.Movie;
import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.data.jpa.repository.Query;
import org.springframework.data.repository.query.Param;
import org.springframework.stereotype.Repository;

import java.util.List;
import java.util.Optional;

@Repository
public interface UserRatingRepository extends JpaRepository<UserRating, Long> {
    // 根据用户ID和电影ID查找评分记录
    UserRating findByUser_IdAndMovie_Id(Long userId, Long movieId);
    // 根据用户ID查找评分列表
    List<UserRating> findByUser_Id(Long userId);
    
    // 根据电影ID查找评分列表
    List<UserRating> findByMovie_Id(Long movieId);
    
    // 根据用户和电影查找评分记录
    Optional<UserRating> findByUserAndMovie(User user, Movie movie);
    
    // 检查用户是否对某部电影评过分
    boolean existsByUser_IdAndMovie_Id(Long userId, Long movieId);
    
    // 计算电影的平均评分
    @Query("SELECT AVG(ur.rating) FROM UserRating ur WHERE ur.movie.id = :movieId")
    Double calculateAverageRatingByMovieId(@Param("movieId") Long movieId);
    
    // 统计电影的评分数量
    long countByMovie_Id(Long movieId);
    
    // 根据用户ID和电影ID删除评分
    void deleteByUser_IdAndMovie_Id(Long userId, Long movieId);
}